from typing import Any

import torch
from torch import Tensor
from torch.nn import functional


class RegressionModule(torch.nn.Module):
    def _forward_unimplemented(self, *input: Any) -> None:
        pass

    def __init__(self):
        super(RegressionModule, self).__init__()
        self.linear = torch.nn.Linear(4, 6)
        self.result = torch.nn.Linear(6, 3)

    def forward(self, x:Tensor):
        std1=x.std(dim=0).sum()
        x = self.linear(x)
        std2=x.std(dim=0).sum()
        x = functional.relu(x)
        x = self.result(x)
        return x
